package cn.isjinhao.se.script;

import java.io.File;

/**
 * @Author ISJINHAO
 * @Date 2021/7/1 17:45
 */
public class FileWrapper {

    private File file;

    private FileType fileType;

    /**
     * 在 xlsx 和 xls 文件中使用
     */
    private int sheetNo;


    public File getFile() {
        return file;
    }

    public void setFile(File file) {
        this.file = file;
        if (file != null) {
            String name = file.getName();
            if (name.endsWith(".xlsx")) {
                this.fileType = FileType.XLSX;
            } else if (name.endsWith(".xls")) {
                this.fileType = FileType.XLS;
            } else if (name.endsWith(".sql")) {
                this.fileType = FileType.SQL;
            }
        }
    }

    public FileType getFileType() {
        return fileType;
    }

    public void setFileType(FileType fileType) {
        this.fileType = fileType;
    }

    public int getSheetNo() {
        return sheetNo;
    }

    public void setSheetNo(int sheetNo) {
        this.sheetNo = sheetNo;
    }


    public enum FileType {
        XLSX, SQL, XLS
    }

}
